Minor change to reflect changes in imported modules