Fix mistake and add test to build