Fixed the bug

Turns out it was an easy fix.