diff --git a/js/admin.js b/js/admin.ts
similarity index 100%
rename from js/admin.js
rename to js/admin.ts
diff --git a/js/forum.js b/js/forum.ts
similarity index 100%
rename from js/forum.js
rename to js/forum.ts
diff --git a/js/src/@types/shims.d.ts b/js/src/@types/shims.d.ts
new file mode 100644
index 0000000..fe87424
--- /dev/null
+++ b/js/src/@types/shims.d.ts
@@ -0,0 +1,9 @@
+declare module 'flarum/common/models/User' {
+  export default interface User {
+    followed(): boolean;
+    blocksFollow(): boolean;
+    canBeFollowed(): boolean;
+    followingCount(): number;
+    followerCount(): number;
+  }
+}
diff --git a/js/src/forum/addFollowControls.js b/js/src/forum/addFollowControls.js
index cb59e10..e5649c4 100644
--- a/js/src/forum/addFollowControls.js
+++ b/js/src/forum/addFollowControls.js
@@ -51,7 +51,7 @@ export default function addFollowControls() {
 
     items.add(
       'follow',
-      <Button icon={icon} onclick={openFollowLevelModal.bind(this, user)}>
+      <Button className="Button" icon={icon} onclick={openFollowLevelModal.bind(this, user)}>
         {app.translator.trans(`ianm-follow-users.forum.user_controls.${user.followed() ? 'change_button' : 'follow_button'}`)}
       </Button>
     );
@@ -70,7 +70,7 @@ export default function addFollowControls() {
     }
 
     const followButton = (
-      <Button className="Button" icon="fas fa-user-friends" onclick={openFollowLevelModal.bind(this, user)}>
+      <Button className="Button Button--follow-profile" icon="fas fa-user-friends" onclick={openFollowLevelModal.bind(this, user)}>
         {user.followed()
           ? app.translator.trans(`ianm-follow-users.forum.badge.label.${user.followed()}`)
           : app.translator.trans('ianm-follow-users.forum.user_controls.follow_button')}