Fixed tests

Signed-off-by: Daniel Bondergaard <danielboendergaard@gmail.com>