Added spec for return