Unbroke some code