Minor module tweak