Fix this spec also