Fixed tiny bug, seems to all be working now