Last commit was likely broken

Fixed errors from last commit