Got tests working. Turns out tests were wrong haha