Add check to make sure chaining works as expected