quick fix, tests to come