Fix getSource() method issues