Fixing test so build currently passes