Fixed dataset failing test