fix returning variable