Fix wrong tests