fixes broken tests (that's an embarassment!)