fix: images cannot scale properly if it's smaller than window

1 files changed, 7 insertions, 6 deletions

diff --git a/src/animation.js b/src/animation.js
index b62fc21..6952a72 100644
--- a/src/animation.js
+++ b/src/animation.js
@@ -10,13 +10,14 @@ export default class Animation {
     if (element.naturalWidth > ww) {
       finals.width = ww;
       finals.height = ww / ratio;
-      if (finals.height > wh) {
-        finals.height = wh;
-        finals.width = wh * ratio;
-      }
     } else {
-      finals.width = element.offsetWidth;
-      finals.height = element.offsetHeight;
+      finals.width = element.naturalWidth;
+      finals.height = element.naturalHeight;
+    }
+
+    if (finals.height > wh) {
+      finals.height = wh;
+      finals.width = wh * ratio;
     }
 
     finals.top = (wh - finals.height) / 2 + window.scrollY + margin;