Finally got tests working