Fixed extend method to return