Maybe i should be doing integration tests?