Fixes broken wip code