Fixes broken code from last push